maximum element in array in python